package com.itheima;

import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.junit.Test;

import java.io.FileOutputStream;
import java.io.IOException;

public class POI_test {
    //使用poi读取excel文件
    @Test
    public void test1() throws IOException {
        //创建工作簿
        XSSFWorkbook work = new XSSFWorkbook("C:\\Users\\Administrator\\Desktop\\aaa.xlsx");
        //获取工作表，既可以根据工作表的顺序获取， 也可以根据工作表的名称互殴去
        XSSFSheet sheet = work.getSheetAt(0);
        //遍历工作表获得行对象
        for (Row row : sheet) {
            //获得单元格
            for (Cell cell : row) {
                String value = cell.getStringCellValue();
                System.out.println(value);
            }
        }
        work.close();
    }

    @Test
    public void test2() throws IOException {
        //创建工作簿
        XSSFWorkbook work = new XSSFWorkbook("C:\\Users\\Administrator\\Desktop\\aaa.xlsx");
        //获取工作表，既可以根据工作表的顺序获取， 也可以根据工作表的名称互殴去
        XSSFSheet sheet = work.getSheetAt(0);

        //获取当前工作表最后一行的行号， 行号从0开始
        int lastRowNum = sheet.getLastRowNum();
        for (int i = 0; i <= lastRowNum; i++) {
            XSSFRow row = sheet.getRow(i);//根据行号获取每一行
            short lastCellNum = row.getLastCellNum();//获取当前行最后一个单元格索引
            for(short j=0;j<lastCellNum;j++){
                String value = row.getCell(j).getStringCellValue();//根据单元格索引获得单元格对象
                System.out.println(value);
            }
        }
        work.close();
    }

    @Test
    public void test3() throws IOException {
        //在内存中创建一个Excle文件
        XSSFWorkbook workbook = new XSSFWorkbook();
        //创建工作表， 指定工作表名称
        XSSFSheet sheet = workbook.createSheet("传智播客");

        //创建行，0表示第一行
        XSSFRow row = sheet.createRow(0);
        //创建单元格， 0表示第一个单元格
        row.createCell(0).setCellValue("编号");
        row.createCell(1).setCellValue("名称");
        row.createCell(2).setCellValue("年龄");

        XSSFRow row1 = sheet.createRow(1);
        row1.createCell(0).setCellValue("1");
        row1.createCell(1).setCellValue("小明");
        row1.createCell(2).setCellValue("10");

        XSSFRow row2 = sheet.createRow(2);
        row2.createCell(0).setCellValue("2");
        row2.createCell(1).setCellValue("小王");
        row2.createCell(2).setCellValue("20");

        //通过输出流将workbook对象下载到磁盘
        FileOutputStream out = new FileOutputStream("C:\\Users\\Administrator\\Desktop\\bbb.xlsx");
        workbook.write(out);
        out.flush();
        out.close();
        workbook.close();
    }
}
